package io.smallrye.graphql.tests.resolvers;

import static org.assertj.core.api.BDDAssertions.then;

import java.net.URL;
import java.util.HashMap;
import java.util.Map;

import org.eclipse.microprofile.graphql.GraphQLApi;
import org.eclipse.microprofile.graphql.Mutation;
import org.eclipse.microprofile.graphql.Name;
import org.eclipse.microprofile.graphql.Query;
import org.eclipse.microprofile.graphql.Source;
import org.jboss.arquillian.container.test.api.Deployment;
import org.jboss.arquillian.container.test.api.RunAsClient;
import org.jboss.arquillian.junit5.ArquillianExtension;
import org.jboss.arquillian.test.api.ArquillianResource;
import org.jboss.shrinkwrap.api.ShrinkWrap;
import org.jboss.shrinkwrap.api.asset.StringAsset;
import org.jboss.shrinkwrap.api.spec.WebArchive;
import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.extension.ExtendWith;

import io.smallrye.graphql.api.federation.Target;
import io.smallrye.graphql.tests.GraphQLAssured;

@ExtendWith(ArquillianExtension.class)
@RunAsClient
public class ResolverTest {
    @Deployment
    public static WebArchive deployment() {
        return ShrinkWrap.create(WebArchive.class, "mutation-target-test.war")
                .addAsResource(new StringAsset("mp.graphql.showErrorMessage=*"), "META-INF/microprofile-config.properties")
                .addClasses(SuperHero.class, Team.class, SuperHeroResource.class);
    }

    public static class SuperHero {
        private String id;
        private String name;

        @SuppressWarnings("unused")
        public SuperHero() {
        }

        public SuperHero(String id, String name) {
            this.id = id;
            this.name = name;
        }

        public void setId(String id) {
            this.id = id;
        }

        public String getId() {
            return id;
        }

        public String getName() {
            return name;
        }

        public void setName(String name) {
            this.name = name;
        }
    }

    public static class Team {
        private String id;
        private String name;

        @SuppressWarnings("unused")
        public Team() {
        }

        public Team(String id, String name) {
            this.id = id;
            this.name = name;
        }

        public void setId(String id) {
            this.id = id;
        }

        public String getId() {
            return id;
        }

        public String getName() {
            return name;
        }

        public void setName(String name) {
            this.name = name;
        }
    }

    @GraphQLApi
    public static class SuperHeroResource {
        private static final Map<String, SuperHero> SUPER_HEROES = Map.of(
                "iron-man", new SuperHero("iron-man", "Iron Man"),
                "spider-man", new SuperHero("spider-man", "Spider Man")
        );
        private static final Map<String, Team> TEAMS = Map.of(
                "avengers", new Team("avengers", "Avengers")
        );
        private static final Map<String, String> SUPER_HERO_TEAMS = new HashMap<>(Map.of(
                "iron-man", "avengers"
        ));

        @Query
        public SuperHero get(String id) {
            var superHero = SUPER_HEROES.get(id);
            if (superHero == null) {
                throw new IllegalArgumentException("unknown super hero; only have " + SUPER_HEROES.keySet());
            }
            return superHero;
        }

        public Team team(@Source SuperHero superHero) {
            var teamId = SUPER_HERO_TEAMS.get(superHero.id);
            if (teamId == null) {
                return null;
            }
            var team = TEAMS.get(teamId);
            if (team == null) {
                throw new IllegalArgumentException("unknown team; only have " + TEAMS.keySet());
            }
            return team;
        }

        @Mutation
        public SuperHero update(@Name("superHero") SuperHero patch) {
            var superHero = get(patch.getId());
            if (patch.getName() != null) {
                superHero.setName(patch.getName());
            }
            return superHero;
        }

        // TODO this should be unnecessary (see blow)
        @Mutation
        public void updateTeam(String superHeroId, String teamId) {
            SUPER_HERO_TEAMS.put(superHeroId, teamId);
        }

        @SuppressWarnings("unused")
        public void updateTeam(@Target SuperHero superHero, String teamId) {
            SUPER_HERO_TEAMS.put(superHero.getId(), teamId);
        }
    }

    @ArquillianResource
    URL testingURL;
    GraphQLAssured gql;

    @BeforeEach
    void setUp() {
        gql = new GraphQLAssured(testingURL);
    }

    @Test
    public void shouldGetSuperHeroWithTeam() {
        var response = gql.post("""
                query {
                  get(id: "iron-man") {
                    name
                    team { name }
                  }
                }""");

        then(response).isEqualTo("""
                {"data":{"get":{"name":"Iron Man","team":{"name":"Avengers"}}}}""");
    }

    @Test
    public void shouldUpdateTeam() {
        // TODO this should be unnecessary if we could provide the rating directly with the update mutation below (commented out)
        then(gql.post("""
                mutation{updateTeam(superHeroId:"spider-man" teamId:"avengers")}""")
        ).isEqualTo("""
                {"data":{"updateTeam":null}}""");

        var response = gql
                .post("""
                        mutation {
                          update (superHero: {
                            id: "spider-man"
                            name: "Ultimate Spiderman"
                            # teamId: "avengers" # <- this is the @Target field
                          }) {
                            name
                            team {name}
                          }
                        }""");

        then(response).isEqualTo("""
                {"data":{"update":{"name":"Ultimate Spiderman","team":{"name":"Avengers"}}}}""");
        then(gql.post("""
                query { get(id:"spider-man") { team { name }}}""")
        ).isEqualTo("""
                {"data":{"get":{"team":{"name":"Avengers"}}}}""");
    }
}